package models

import (
	"fmt"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

var (
	db *gorm.DB
)

func init(){   //初始化数据库连接,创建表结构
	//使用dsn连接到数据库，grom自带的数据库池
	//账号:密码@连接方式(ip地址:端口号)/数据库？语言方式，时区（未设置时区的话采用8小时制度）
	var err error
	dsn := "root:root@tcp(127.0.0.1:3306)/testgorm?charset=utf8mb4&parseTime=True&loc=Local"
	db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{}) //使用mysq连接数据库，第二个参数可以增加更多的配置(可有可无)
	if err != nil {
		fmt.Println(err)
	}
	_ = db.AutoMigrate(&User{}, &Node{},&Message{}) //创建User表,创建Node表,Message表
	//判断如果数据库没有值，则增加管理员
    if res:=db.Find(&User{});res.RowsAffected==0&&res.Error==nil{
    	db.Create(&User{
			Name:"admin",
			Email:"1102394156@qq.com",
			Pwd:"123123",
			Head:"/static/images/item.png",
			Role: 0,
		})
	}
}

